diff --git a/main.py b/main.py
index 325fc59..66a7d9f 100644
--- a/main.py
+++ b/main.py
@@ -19,7 +19,6 @@ def chat(
     run_prefix: str = typer.Option("", help="run prefix, if you want to run multiple variants of the same project and later compare them"),
     model: str = "gpt-4",
     temperature: float = 0.1,
-    max_tokens: int = 8192,
 ):
 
     if project_path is None:
@@ -32,7 +31,6 @@ def chat(
     ai = AI(
         model=model,
         temperature=temperature,
-        max_tokens=max_tokens,
     )
 
     dbs = DBs(
@@ -50,4 +48,4 @@ def chat(
 
 
 if __name__ == "__main__":
-    app()
\ No newline at end of file
+    app()
diff --git a/steps.py b/steps.py
index 970a2db..a5028f4 100644
--- a/steps.py
+++ b/steps.py
@@ -1,6 +1,7 @@
 from ai import AI
 from chat_to_files import to_files
 from db import DBs
+import json
 
 
 def setup_sys_prompt(dbs):