diff --git a/README.md b/README.md
index e6dea7ca..66311f61 100644
--- a/README.md
+++ b/README.md
@@ -571,9 +571,7 @@ import os
 from dotenv import load_dotenv
 
 # Import the OpenAIChat model and the Agent struct
-from swarms.models import OpenAIChat
-from swarms.structs import Agent
-from swarms.structs.swarm_net import SwarmNetwork
+from swarms import OpenAIChat, Agent, SwarmNetwork
 
 # Load the environment variables
 load_dotenv()
@@ -615,7 +613,6 @@ out = swarmnet.run_many_agents(
 )
 print(out)
 
-
 ```
 
 ---
diff --git a/pyproject.toml b/pyproject.toml
index 34339640..33e9fb54 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms"
-version = "3.1.0"
+version = "3.1.1"
 description = "Swarms - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <kye@apac.ai>"]
diff --git a/swarm_network.py b/swarm_network.py
index ba96a934..de9c53b6 100644
--- a/swarm_network.py
+++ b/swarm_network.py
@@ -3,9 +3,7 @@ import os
 from dotenv import load_dotenv
 
 # Import the OpenAIChat model and the Agent struct
-from swarms.models import OpenAIChat
-from swarms.structs import Agent
-from swarms.structs.swarm_net import SwarmNetwork
+from swarms import OpenAIChat, Agent, SwarmNetwork
 
 # Load the environment variables
 load_dotenv()
diff --git a/swarms/swarms/__init__.py b/swarms/swarms/__init__.py
index 7f1d806a..ed39abfb 100644
--- a/swarms/swarms/__init__.py
+++ b/swarms/swarms/__init__.py
@@ -2,6 +2,7 @@ from swarms.structs.autoscaler import AutoScaler
 from swarms.swarms.model_parallizer import ModelParallelizer
 from swarms.swarms.multi_agent_collab import MultiAgentCollaboration
 from swarms.swarms.base import AbstractSwarm
+# from swarms.swarms.team import Team
 
 __all__ = [
     "AutoScaler",