diff --git a/mine.lua b/mine.lua
index 308dae1..2619f89 100644
--- a/mine.lua
+++ b/mine.lua
@@ -1,11 +1,15 @@
 local robot = require("robot")
 
-local function mine()
-  robot.swing()
-  robot.up()
-  robot.swing()
-  robot.down()
-  robot.forward()
+local arg = ...
+
+local function mine(count)
+  for i = 1, count do
+    robot.swing()
+    robot.up()
+    robot.swing()
+    robot.down()
+    robot.forward()
+  end
 end
 
-mine()
+mine(arg)