diff --git a/vision_agent/agent/agent_coder.py b/vision_agent/agent/agent_coder.py
index 401029fb..38b9515e 100644
--- a/vision_agent/agent/agent_coder.py
+++ b/vision_agent/agent/agent_coder.py
@@ -82,6 +82,17 @@ def fix_bugs(code: str, tests: str, result: str, feedback: str, model: LLM) -> s
 
 
 class AgentCoder(Agent):
+    """AgentCoder is based off of the AgentCoder paper https://arxiv.org/abs/2312.13010
+    and it's open source code https://github.com/huangd1999/AgentCoder with some key
+    differences. AgentCoder comprises of 3 components: a coder agent, a tester agent,
+    and an executor. The tester agents writes code to test the code written by the coder
+    agent, but in our case because we are solving a vision task it's difficult to write
+    testing code. We instead have the tester agent write code to visualize the output
+    of the code written by the coder agent. If the code fails, we pass it back to the
+    coder agent to fix the bug, if it succeeds we pass it to a visual tester agent, which
+    is an LMM model like GPT4V, to visually inspect the output and make sure it looks
+    good."""
+
     def __init__(
         self,
         coder_agent: Optional[LLM] = None,
diff --git a/vision_agent/agent/execution.py b/vision_agent/agent/execution.py
index fb9d1649..fe111546 100644
--- a/vision_agent/agent/execution.py
+++ b/vision_agent/agent/execution.py
@@ -1,3 +1,6 @@
+"""This code is based off of code from CodeGeeX https://github.com/THUDM/CodeGeeX
+
+"""
 import contextlib
 import faulthandler
 import io