Allow Saving of assignments

main.py

@@ -1,14 +1,28 @@
 import firepup650 as fp
 
+keys = {"Create new assignment": False}
+
+try:
+    from assignments import keys
+except:
+    with open("assignments.py", "w") as file:
+        file.write(f"keys = {keys}")
+
 fp.replitCursor = fp.bcolors.REPLIT + "> " + fp.bcolors.RESET
 
 answerKey = []
 answer = "0"
 answerCount = 0
+key = False
 
+if len(keys) > 1:
+    key = fp.menu(keys, "Please select an assignment to grade")
 fp.clear()
-
-while not False:
+if key:
+    answerKey = key
+    answerCount = len(key)
+else:
+    while not False:
         answer = fp.replitInput(
             f"Please input the answer to question #{answerCount + 1}, empty answer to submit answer key"
         ).upper()
@@ -23,6 +37,10 @@ while not False:
                 continue
         answerCount += 1
         answerKey.append(answer)
+    name = fp.replitInput("Please enter the name of this assignment")
+    keys[name] = answerKey
+    with open("assignments.py", "w") as file:
+        file.write(f"keys = {keys}")
 
 fp.clear()