mine n

mine.lua

@@ -1,11 +1,15 @@
 local robot = require("robot")
 
-local function mine()
+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)