diff --git a/browserpilot/agents/compilers/instruction_compiler.py b/browserpilot/agents/compilers/instruction_compiler.py
index a7c8dfb..f9f31c0 100644
--- a/browserpilot/agents/compilers/instruction_compiler.py
+++ b/browserpilot/agents/compilers/instruction_compiler.py
@@ -6,6 +6,7 @@
 import logging
 import traceback
 import os
+import re
 
 from typing import Dict, List, Union
 
@@ -297,7 +298,6 @@ def get_completion(
                     frequency_penalty=0,
                     presence_penalty=0,
                     temperature=temperature,
-                    stop=stop,
                 )
                 text = response.choices[0].message.content
             else:
@@ -326,6 +326,9 @@ def get_completion(
         except Exception:
             traceback.print_exc()
 
+        # Extract python code
+        text = self._extract_python_code(text)
+
         # Add to cache.
         self.api_cache[prompt] = text
 
@@ -406,6 +409,15 @@ def save_compiled_instructions(self, filename):
             elif filename.endswith(".yaml"):
                 yaml.dump(to_dump, f)
 
+    def _extract_python_code(self, input_string: str) -> str:
+        pattern = r'```python\n(.*?)```'
+        match = re.search(pattern, input_string, re.DOTALL)
+
+        if match:
+            return match.group(1)
+        else:
+            return input_string
+
 
 if __name__ == "__main__":
     import pprint
diff --git a/examples.py b/examples.py
index a64b45a..cdd9dad 100644
--- a/examples.py
+++ b/examples.py
@@ -1,7 +1,6 @@
 import click
 
 from browserpilot.agents.gpt_selenium_agent import GPTSeleniumAgent
-# from browserpilot.agents.goal_agent import GoalAgent
 
 
 # Set up multiple command CLI.
@@ -24,6 +23,7 @@ def selenium(instructions, chromedriver_path, model, memory_folder, debug, outpu
             chromedriver_path,
             instruction_output_file=output,
             model_for_instructions=model,
+            model_for_responses=model,
             memory_folder=memory_folder,
             debug=debug,
             retry=True,
diff --git a/prompts/examples/github_issue.yaml b/prompts/examples/github_issue.yaml
new file mode 100644
index 0000000..f379092
--- /dev/null
+++ b/prompts/examples/github_issue.yaml
@@ -0,0 +1,18 @@
+instructions:
+- BEGIN_FUNCTION open_github_issue
+- Go to Google.com
+- Find all textareas.
+- Find the first visible textarea.
+- Click on the first visible textarea.
+- Type in "netbox github" and press enter.
+- Wait 2 seconds.
+- Find all anchor elements that link to GitHub.com.
+- Click on the first one.
+- Wait 2 seconds.
+- Click on the anchor element with id 'issues-tab'.
+- Wait 2 seconds.
+- Click on the closed issues to filter them.
+- Wait 2 seconds.
+- END_FUNCTION
+- RUN_FUNCTION open_github_issue
+- Wait for 10 seconds.
\ No newline at end of file