clean exit

src/tea3/cli.py

@@ -190,6 +190,7 @@ def run_exhaustive_xor_cli():
 
 
 def main():
+    while True:
         print("=" * 50)
         print(" Tea3 Model ")
         print("=" * 50)
@@ -199,13 +200,20 @@ def main():
         print("  2) Advanced inspection with forced 0 or 1 bits")
         print("  3) Exhaustive variable-change search")
         print("  4) S box analysis")
-    print("  5) variable XOR")
+        print("  5) Variable XOR")
         print("  6) F31, F32 analysis")
         print("  7) Exhaustive XOR search")
+        print("  0) Exit")
 
-    mode = prompt_choice("Your choice (1, 2, 3, 4, 5, 6 or 7): ", {1, 2, 3, 4, 5, 6, 7})
+        mode = prompt_choice(
+            "Your choice (0-7): ",
+            {0, 1, 2, 3, 4, 5, 6, 7}
+        )
 
-    if mode == 1:
+        if mode == 0:
+            print("\nGoodbye")
+            break
+        elif mode == 1:
             run_classic_cli()
         elif mode == 2:
             run_advanced_cli()
@@ -220,5 +228,7 @@ def main():
         elif mode == 7:
             run_exhaustive_xor_cli()
 
+        input("\nPress Enter to return to the main menu...")
+        print()
 
 main()