iteration

lum.lua

@@ -1,11 +1,6 @@
 local args = { ... }
 
 local action = args[1]
-if action == nil then
-    print("Usage: lum <install:remove>")
-    return
-end
-
 
 if action == "install" then
     package = args[2]
@@ -16,4 +11,17 @@ if action == "install" then
 
     lum.install(package)
     io.write("Done!\n")
+elseif action == "remove" then
+    package = args[2]
+    if package == nil then
+        print("Usage: lum remove <package>")
+        return
+    end
+
+    lum.remove(package)
+    io.write("Done!\n")
+elseif action == "version" then
+    print("Lum version 0.1.0")
+else
+    io.write("Usage: lum <install:remove:version>\n")
 end