mine n

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)