diff --git a/omnimodal_agent.py b/omnimodal_agent.py
new file mode 100644
index 00000000..07c0a76b
--- /dev/null
+++ b/omnimodal_agent.py
@@ -0,0 +1,7 @@
+from swarms import OmniModalAgent, OpenAIChat
+
+llm = OpenAIChat()
+
+agent = OmniModalAgent(llm)
+
+agent.run("Create a video of a swarm of fish")
\ No newline at end of file
diff --git a/swarms/agents/__init__.py b/swarms/agents/__init__.py
index d3f76509..11832e55 100644
--- a/swarms/agents/__init__.py
+++ b/swarms/agents/__init__.py
@@ -1,23 +1,10 @@
 """Agent Infrastructure, models, memory, utils, tools"""
 
-
-###########
-# from swarms.tools.base import BaseTool, Tool, StructuredTool, ToolWrapper, BaseToolSet, ToolCreator, GlobalToolsCreator, SessionToolsCreator, ToolsFactory
-# from swarms.tools.autogpt import pushd, process_csv, async_load_playwright, run_async, browse_web_page, WebpageQATool, web_search, query_website_tool
-# from swarms.tools.exit_conversation import ExitConversation
-
-# from swarms.tools.models import MaskFormer, ImageEditing, InstructPix2Pix, Text2Image, VisualQuestionAnswering, ImageCaptioning
-# from swarms.tools.file_mangagement import read_tool, write_tool, list_tool
-# from swarms.tools.requests import RequestsGet
-
-# from swarms.tools.developer import Terminal, CodeEditor
-
-
 #agents
 # from swarms.agents.profitpilot import ProfitPilot
 from swarms.agents.aot import AoTAgent
-# from swarms.agents.omni_modal_agent import OmniModalAgent
 from swarms.agents.multi_modal_visual_agent import MultiModalAgent
+from swarms.agents.omni_modal_agent import OmniModalAgent