diff --git a/py/api.py b/py/api.py
index 8ed3afb..413f951 100644
--- a/py/api.py
+++ b/py/api.py
@@ -2,6 +2,7 @@ import requests
 import json
 from transformers import AutoTokenizer, LlamaForCausalLM
 
+
 class API:
     @staticmethod
     def process_text_transformers(prompt, model):
@@ -46,4 +47,4 @@ class API:
             answer = self.process_text_transformers(message, "meta-llama/Meta-Llama-3.1-8B")
         else:
             return "Invalid choice"
-        return answer
\ No newline at end of file
+        return answer
diff --git a/py/simple_chat.py b/py/simple_chat.py
index 3bb0c16..0c3165f 100644
--- a/py/simple_chat.py
+++ b/py/simple_chat.py
@@ -1,20 +1,24 @@
 from api import API
 
-chat1 = API()
 
-while True:
-    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
+class CLIChat:
+    @staticmethod
+    def chat():
+        chat1 = API()
+
+        while True:
+            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("Invalid model.")
+                continue
+            while True:
+                print("")
+                print("Ask a question")
+                input_text = input()
+                if input_text == "change":
+                    break
+                output_text = chat1.send_message(input_text, model)
+                print(output_text)
diff --git a/py/static/index.html b/py/static/index.html
index 087940a..d688702 100644
--- a/py/static/index.html
+++ b/py/static/index.html
@@ -4,7 +4,7 @@
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <link rel="stylesheet" href="styles.css">
+    <link rel="stylesheet" href="/static/styles.css">
     <title>AI Assistant</title>
 </head>
 
@@ -81,10 +81,10 @@
         <!-- Input section: Where user input is provided -->
         <div class="input">
             <input type="text" placeholder="Type your message here..." />
-            <button><img src="img/microphone.svg" alt="microphone"></button>
+            <button><img src="/static/img/microphone.svg" alt="microphone"></button>
             <button>Send</button>
         </div>
     </div>
 </body>
 
-</html>
\ No newline at end of file
+</html>
diff --git a/py/web_flask.py b/py/web_flask.py
index 3cc2681..ce616ed 100644
--- a/py/web_flask.py
+++ b/py/web_flask.py
@@ -1,11 +1,14 @@
-from flask import Flask
-
-app = Flask(__name__)
+from flask import Flask, send_from_directory
 
 
-@app.route('/')
-def index():
-    return app.send_static_file('index.html')
+class WebHost:
+    @staticmethod
+    def main_page():
+        app = Flask(__name__)
 
-if __name__ == '__main__':
-    app.run()
+        @app.route('/')
+        def index():
+            return app.send_static_file('index.html')
+
+        if __name__ == '__main__':
+            app.run(debug=True)
