diff --git a/swarms/__init__.py b/swarms/__init__.py
index 581023f1..610b8525 100644
--- a/swarms/__init__.py
+++ b/swarms/__init__.py
@@ -26,3 +26,10 @@ from swarms.structs.workflow import Workflow
 from swarms.swarms.dialogue_simulator import DialogueSimulator
 from swarms.swarms.autoscaler import AutoScaler
 from swarms.swarms.orchestrate import Orchestrator
+
+
+#agents
+from swarms.swarms.profitpilot import ProfitPilot
+from swarms.aot import AoTAgent
+from swarms.agents.multi_modal_agent import MultiModalVisualAgent
+from swarms.agents.omni_modal_agent import OmniModalAgent
\ No newline at end of file
diff --git a/swarms/agents/multi_modal_agent.py b/swarms/agents/multi_modal_agent.py
index 4534d0fe..be6831cd 100644
--- a/swarms/agents/multi_modal_agent.py
+++ b/swarms/agents/multi_modal_agent.py
@@ -1,8 +1,10 @@
-# from swarms.workers.multi_modal_workers.multi_modal_agent import MultiModalVisualAgent
 from swarms.workers.multi_modal_workers.multi_modal_agent import MultiModalVisualAgent
 
 class MultiModalVisualAgent:
-    def __init__(self, agent: MultiModalVisualAgent):
+    def __init__(
+            self, 
+            agent: MultiModalVisualAgent
+        ):
         self.agent = agent
     
     def _run(self, text: str) -> str:
diff --git a/swarms/workers/neural_architecture_search_worker.py b/swarms/agents/neural_architecture_search_worker.py
similarity index 99%
rename from swarms/workers/neural_architecture_search_worker.py
rename to swarms/agents/neural_architecture_search_worker.py
index 73b492bb..018febc2 100644
--- a/swarms/workers/neural_architecture_search_worker.py
+++ b/swarms/agents/neural_architecture_search_worker.py
@@ -1,7 +1,5 @@
 from swarms.agents.aot import AoTAgent
 
-
-
 task = "Create GPT-2"
 
 
diff --git a/swarms/agents/omni_modal_agent.py b/swarms/agents/omni_modal_agent.py
index b30b601b..a88ae191 100644
--- a/swarms/agents/omni_modal_agent.py
+++ b/swarms/agents/omni_modal_agent.py
@@ -1,5 +1,3 @@
-from langchain.tools import tool
-
 from swarms.workers.multi_modal_workers.omni_agent.omni_chat import chat_huggingface
 
 
@@ -13,7 +11,6 @@ class OmniModalAgent:
         self.api_endpoint = api_endpoint
         self.api_type = api_type
 
-    @tool
     def chat(self, data):
         """Chat with omni-modality model that uses huggingface to query for a specific model at run time. Translate text to speech, create images and more"""
         messages = data.get("messages")
diff --git a/swarms/agents/profitpilot.py b/swarms/agents/profitpilot.py
index e4f632b7..9f30127e 100644
--- a/swarms/agents/profitpilot.py
+++ b/swarms/agents/profitpilot.py
@@ -17,7 +17,6 @@ from langchain.text_splitter import CharacterTextSplitter
 from langchain.vectorstores import Chroma
 from pydantic import BaseModel, Field
 
-
 llm = ChatOpenAI(temperature=0.9)
 
 
diff --git a/swarms/workers/multi_modal_workers/multi_modal_agent.py b/swarms/workers/multi_modal_workers/multi_modal_agent.py
index bd989422..1515b25f 100644
--- a/swarms/workers/multi_modal_workers/multi_modal_agent.py
+++ b/swarms/workers/multi_modal_workers/multi_modal_agent.py
@@ -1559,11 +1559,11 @@ class MultiModalVisualAgent:
     def clear_memory(self):
         self.memory.clear()
 
-if __name__ == '__main__':
-    if not os.path.exists("checkpoints"):
-        os.mkdir("checkpoints")
-    parser = argparse.ArgumentParser()
-    parser.add_argument('--load', type=str, default="ImageCaptioning_cuda:0,Text2Image_cuda:0")
-    args = parser.parse_args()
-    load_dict = {e.split('_')[0].strip(): e.split('_')[1].strip() for e in args.load.split(',')}
-    agent = MultiModalVisualAgent(load_dict=load_dict)
\ No newline at end of file
+# if __name__ == '__main__':
+#     if not os.path.exists("checkpoints"):
+#         os.mkdir("checkpoints")
+#     parser = argparse.ArgumentParser()
+#     parser.add_argument('--load', type=str, default="ImageCaptioning_cuda:0,Text2Image_cuda:0")
+#     args = parser.parse_args()
+#     load_dict = {e.split('_')[0].strip(): e.split('_')[1].strip() for e in args.load.split(',')}
+#     agent = MultiModalVisualAgent(load_dict=load_dict)
\ No newline at end of file