diff --git a/vision_agent/agent/agent_coder.py b/vision_agent/agent/agent_coder.py
index 998c0d8c..fca9ea64 100644
--- a/vision_agent/agent/agent_coder.py
+++ b/vision_agent/agent/agent_coder.py
@@ -4,6 +4,7 @@
 import sys
 from pathlib import Path
 from typing import Dict, List, Optional, Union
+
 from rich.console import Console
 from rich.syntax import Syntax
 
@@ -169,10 +170,14 @@ def chat(
         for _ in range(self.max_turns):
             code = write_program(question, feedback, self.coder_agent)
             if self.verbose:
-                _CONSOLE.print(Syntax(code, "python", theme="gruvbox-dark", line_numbers=True))
+                _CONSOLE.print(
+                    Syntax(code, "python", theme="gruvbox-dark", line_numbers=True)
+                )
             debug = write_debug(question, code, feedback, self.tester_agent)
             if self.verbose:
-                _CONSOLE.print(Syntax(debug, "python", theme="gruvbox-dark", line_numbers=True))
+                _CONSOLE.print(
+                    Syntax(debug, "python", theme="gruvbox-dark", line_numbers=True)
+                )
             results = execute_tests(code, debug)
             _LOGGER.info(
                 f"execution results: passed: {results['passed']}\n{results['result']}"
@@ -183,7 +188,9 @@ def chat(
                     code, debug, results["result"].strip(), feedback, self.coder_agent  # type: ignore
                 )
                 if self.verbose:
-                    _CONSOLE.print(Syntax(code, "python", theme="gruvbox-dark", line_numbers=True))
+                    _CONSOLE.print(
+                        Syntax(code, "python", theme="gruvbox-dark", line_numbers=True)
+                    )
             else:
                 # TODO: Sometimes it prints nothing, so we need to handle that case
                 # TODO: The visual agent reflection does not work very well, needs more testing
diff --git a/vision_agent/agent/vision_agent_v2.py b/vision_agent/agent/vision_agent_v2.py
index 9bb39c1b..6053994f 100644
--- a/vision_agent/agent/vision_agent_v2.py
+++ b/vision_agent/agent/vision_agent_v2.py
@@ -379,7 +379,9 @@ def chat_with_workflow(
                 self.long_term_memory,
                 self.verbosity,
             )
-            success = all(task["success"] if "success" in task else False for task in plan)
+            success = all(
+                task["success"] if "success" in task else False for task in plan
+            )
             working_memory.update(working_memory_i)
 
             if not success:
diff --git a/vision_agent/tools/tools_v2.py b/vision_agent/tools/tools_v2.py
index d0845f1c..2cd33df2 100644
--- a/vision_agent/tools/tools_v2.py
+++ b/vision_agent/tools/tools_v2.py
@@ -1,8 +1,8 @@
 import inspect
 import io
+import json
 import logging
 import tempfile
-import json
 from importlib import resources
 from pathlib import Path
 from typing import Any, Callable, Dict, List, Tuple, Union, cast
diff --git a/vision_agent/utils/sim.py b/vision_agent/utils/sim.py
index ad50149a..614fdefa 100644
--- a/vision_agent/utils/sim.py
+++ b/vision_agent/utils/sim.py
@@ -56,7 +56,9 @@ def save(self, sim_file: Union[str, Path]) -> None:
         df = df.drop("embs", axis=1)
         df.to_csv(sim_file / "df.csv", index=False)
 
-    def top_k(self, query: str, k: int = 5, thresh: Optional[float] = None) -> Sequence[Dict]:
+    def top_k(
+        self, query: str, k: int = 5, thresh: Optional[float] = None
+    ) -> Sequence[Dict]:
         """Returns the top k most similar items to the query.
 
         Parameters: