diff --git a/scripts/ai_functions.py b/scripts/ai_functions.py
index 0bd6c139..b79c4dbc 100644
--- a/scripts/ai_functions.py
+++ b/scripts/ai_functions.py
@@ -3,22 +3,26 @@ import json
 import openai
 
 
+# This is a magic function that can do anything with no-code. See
+# https://github.com/Torantulino/AI-Functions for more info.
 def call_ai_function(function, args, description, model="gpt-4"):
     # parse args to comma seperated string
     args = ", ".join(args)
-    messages = [{"role": "system",
-                 "content": f"You are now the following python function: ```# {description}\n{function}```\n\nOnly respond with your `return` value."},
-                {"role": "user",
-                 "content": args}]
+    messages = [
+        {
+            "role": "system",
+            "content": f"You are now the following python function: ```# {description}\n{function}```\n\nOnly respond with your `return` value.",
+        },
+        {"role": "user", "content": args},
+    ]
 
     response = openai.ChatCompletion.create(
-        model=model,
-        messages=messages,
-        temperature=0
+        model=model, messages=messages, temperature=0
     )
 
     return response.choices[0].message["content"]
 
+
 # Evaluating code
 
 
@@ -33,8 +37,11 @@ def evaluate_code(code: str) -> List[str]:
 
 # Improving code
 
+
 def improve_code(suggestions: List[str], code: str) -> str:
-    function_string = "def generate_improved_code(suggestions: List[str], code: str) -> str:"
+    function_string = (
+        "def generate_improved_code(suggestions: List[str], code: str) -> str:"
+    )
     args = [json.dumps(suggestions), code]
     description_string = """Improves the provided code based on the suggestions provided, making no other changes."""
 
@@ -44,8 +51,11 @@ def improve_code(suggestions: List[str], code: str) -> str:
 
 # Writing tests
 
+
 def write_tests(code: str, focus: List[str]) -> str:
-    function_string = "def create_test_cases(code: str, focus: Optional[str] = None) -> str:"
+    function_string = (
+        "def create_test_cases(code: str, focus: Optional[str] = None) -> str:"
+    )
     args = [code, json.dumps(focus)]
     description_string = """Generates test cases for the existing code, focusing on specific areas if required."""
 
diff --git a/scripts/keys.py b/scripts/keys.py
index 6dcf1e2a..1cd439cd 100644
--- a/scripts/keys.py
+++ b/scripts/keys.py
@@ -1,2 +1,5 @@
-OPENAI_API_KEY="YOUR-OPENAI-KEY" # Get yours from:  https://beta.openai.com/account/api-keys
-ELEVENLABS_API_KEY="YOUR-ELEVENLABS-KEY" # To access your ElevenLabs API key, head to https://elevenlabs.io, you can view your xi-api-key using the 'Profile' tab on the website.
+# Get yours from:  https://beta.openai.com/account/api-keys
+OPENAI_API_KEY = "YOUR-OPENAI-KEY"
+# To access your ElevenLabs API key, head to https://elevenlabs.io, you
+# can view your xi-api-key using the 'Profile' tab on the website.
+ELEVENLABS_API_KEY = "YOUR-ELEVENLABS-KEY"