diff --git a/example.py b/example.py
index a0e85d52..c0fad2a1 100644
--- a/example.py
+++ b/example.py
@@ -42,13 +42,12 @@ agent = Agent(
     verbose=True,
     streaming_on=True,
     # interactive=True, # Set to False to disable interactive mode
-    # stopping_token="<DONE>",
-    # saved_state_path="accounting_agent.json",
+    saved_state_path="accounting_agent.json",
     # tools=[calculate_profit, generate_report],
     # docs_folder="docs",
     # pdf_path="docs/accounting_agent.pdf",
 )
 
 agent.run(
-    "We're the Swarm Corporation, our total revenue is $100,000 and our total expenses are $50,000."
+    "We're the Swarm Corporation, our total revenue is $100,000 and our total expenses are $50,000, is our revenue good?"
 )
diff --git a/pyproject.toml b/pyproject.toml
index 996c3715..7e0868f4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms"
-version = "5.1.8"
+version = "5.2.0"
 description = "Swarms - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <kye@apac.ai>"]
@@ -52,6 +52,7 @@ PyYAML = "*"
 docstring_parser = "0.16"
 fastapi = "*"
 openai = ">=1.30.1,<2.0"
+termcolor = "*"
 
 
 [tool.poetry.group.lint.dependencies]
diff --git a/swarms/structs/agent.py b/swarms/structs/agent.py
index 20c467b6..2d63340c 100644
--- a/swarms/structs/agent.py
+++ b/swarms/structs/agent.py
@@ -1793,7 +1793,7 @@ class Agent(BaseStructure):
 
         return response
 
-    def stream_response(self, response: str, delay: float = 0.1) -> None:
+    def stream_response(self, response: str, delay: float = 0.01) -> None:
         """
         Streams the response token by token.
 
diff --git a/tests/structs/test_agent_rearrange.py b/tests/structs/test_agent_rearrange.py
index a9e07642..e7d32adb 100644
--- a/tests/structs/test_agent_rearrange.py
+++ b/tests/structs/test_agent_rearrange.py
@@ -1,5 +1,5 @@
 import pytest
-from agent_rearrange import AgentRearrange
+from swarms.structs.rearrange import AgentRearrange
 
 
 # Mocking the Agent class