Modified arguments

koboldgen.py

 
 if __name__ == "__main__":
     args = sys.argv[1:]
-    if "--campaign" in args:
+    if "--campaign" in args or "-c" in args:
         cs = []
         for _ in range(6):
             c = Character()
             k.print()
 
     else:
-        c = Character()
-        c.generate()
-        c.print()
+        if len(args) == 0:
+            c = Character()
+            c.generate()
+            c.print()
+        else:
+            try:
+                num = int(args[0])
+                for _ in range(num):
+                    c = Character()
+                    c.generate()
+                    c.print()
+            except:
+                print(f"Invalid arguments: {', '.join(args)}")
+                c = Character()
+                c.generate()
+                c.print()