diff --git a/py/simple_chat.py b/py/simple_chat.py
index 53203d1..3bb0c16 100644
--- a/py/simple_chat.py
+++ b/py/simple_chat.py
@@ -1,10 +1,20 @@
-from api import api
+from api import API
 
-chat1 = api()
+chat1 = API()
 
 while True:
-    print("Gib deine Frage ein")
-    input_text = input()
-
-    output_text = api.send_message(chat1, input_text)
-    print(output_text)
\ No newline at end of file
+    print("")
+    print("Which AI Model do you want to use? Write as a Number (1-5)")
+    model_input = input()
+    model = int(model_input)
+    if model <=0 or model > 5:
+        print("ungültiges Modell")
+        continue
+    while True:
+        print("")
+        print("Ask a question")
+        inputText = input ()
+        if inputText == "change":
+            break
+        outputText = chat1.send_message(inputText, model)
+        print(outputText)
\ No newline at end of file