From 43759e720b03df98e481efc22a6bc4b7458ac51e Mon Sep 17 00:00:00 2001
From: Sam Hadow <sam.hadow@proton.me>
Date: Mon, 27 Apr 2026 09:41:53 +0200
Subject: [PATCH] cli change

---
 src/tea3/cli.py | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/src/tea3/cli.py b/src/tea3/cli.py
index 4f05682..1807739 100644
--- a/src/tea3/cli.py
+++ b/src/tea3/cli.py
@@ -59,6 +59,17 @@ def run_classic_cli():
     print("\n" + "=" * 50)
     print("Done.")
 
+def run_exhaustive_cli():
+    steps = prompt_int("How many steps? (1–100): ", 1, 100)
+    target_reg = prompt_int("Target register (0–7): ", 0, 7)
+
+    print("-" * 50)
+    run_exhaustive(steps, target_reg)
+
+    print("\n" + "=" * 50)
+    print("Done.")
+
+
 
 def main():
     print("=" * 50)
@@ -74,14 +85,7 @@ def main():
     if mode == 1:
         run_classic_cli()
     else:
-        steps = prompt_int("How many steps? (1–100): ", 1, 100)
-        target_reg = prompt_int("Target register (0–7): ", 0, 7)
-
-        print("-" * 50)
-        run_exhaustive(steps, target_reg)
-
-        print("\n" + "=" * 50)
-        print("Done.")
+        run_exhaustive_cli()
 
 
 main()