diff --git a/swarms/models/__init__.py b/swarms/models/__init__.py
index 44b752ff..30a37a57 100644
--- a/swarms/models/__init__.py
+++ b/swarms/models/__init__.py
@@ -1,3 +1,8 @@
+import sys
+
+log_file = open("errors.txt", "w")
+sys.stderr = log_file
+
 # LLMs
 from swarms.models.anthropic import Anthropic
 from swarms.models.petals import Petals
@@ -18,12 +23,6 @@ from swarms.models.layoutlm_document_qa import LayoutLMDocumentQA
 from swarms.models.gpt4v import GPT4Vision
 from swarms.models.dalle3 import Dalle3
 from swarms.models.distilled_whisperx import DistilWhisperModel
-# from swarms.models.fuyu import Fuyu  # Not working, wait until they update
-
-import sys
-
-log_file = open("errors.txt", "w")
-sys.stderr = log_file
 
 __all__ = [
     "Anthropic",
@@ -44,5 +43,4 @@ __all__ = [
     "WizardLLMStoryTeller",
     "GPT4Vision",
     "Dalle3",
-    # "Fuyu",
 ]
diff --git a/swarms/structs/flow.py b/swarms/structs/flow.py
index 8fcd1d4d..05c1c138 100644
--- a/swarms/structs/flow.py
+++ b/swarms/structs/flow.py
@@ -124,7 +124,7 @@ class Flow:
         dashboard: bool = False,
         agent_name: str = "Flow agent",
         system_prompt: str = FLOW_SYSTEM_PROMPT,
-        # tools: List[BaseTool] = None,
+        tools: List[Any] = None,
         dynamic_temperature: bool = False,
         saved_state_path: Optional[str] = "flow_state.json",
         autosave: bool = False,
@@ -150,7 +150,7 @@ class Flow:
         # The max_loops will be set dynamically if the dynamic_loop
         if self.dynamic_loops:
             self.max_loops = "auto"
-        # self.tools = tools or []
+        self.tools = tools or []
         self.system_prompt = system_prompt
         self.agent_name = agent_name
         self.saved_state_path = saved_state_path