mine n

mine.lua

@@ -1,11 +1,15 @@
 local robot = require("robot")
 
-local function mine()
+local arg = ...
-  robot.swing()
+
-  robot.up()
+local function mine(count)
-  robot.swing()
+  for i = 1, count do
-  robot.down()
+    robot.swing()
-  robot.forward()
+    robot.up()
+    robot.swing()
+    robot.down()
+    robot.forward()
+  end
 end
 
-mine()
+mine(arg)