diff --git a/README.md b/README.md
index 2e3adbd4..d60a110a 100644
--- a/README.md
+++ b/README.md
@@ -34,28 +34,35 @@ Run example in Collab: <a target="_blank" href="https://colab.research.google.co
 
 ```python
 
+import os
+
+from dotenv import load_dotenv
+
+# Import the OpenAIChat model and the Flow struct
 from swarms.models import OpenAIChat
 from swarms.structs import Flow
 
-api_key = ""
+# Load the environment variables
+load_dotenv()
 
-# Initialize the language model, this model can be swapped out with Anthropic, ETC, Huggingface Models like Mistral, ETC
+# Get the API key from the environment
+api_key = os.environ.get("OPENAI_API_KEY")
+
+# Initialize the language model
 llm = OpenAIChat(
-    openai_api_key=api_key,
     temperature=0.5,
+    openai_api_key=api_key,
 )
 
-## Initialize the workflow
-flow = Flow(
-    llm=llm,
-    max_loops=2,
-    dashboard=True,
 
-)
+## Initialize the workflow
+flow = Flow(llm=llm, max_loops=1, dashboard=True)
 
+# Run the workflow on a task
 out = flow.run("Generate a 10,000 word blog on health and wellness.")
 
 
+
 ```
 
 ------
@@ -125,24 +132,32 @@ for task in workflow.tasks:
 ```python
 from swarms.structs import Flow
 from swarms.models.gpt4_vision_api import GPT4VisionAPI
+from swarms.prompts.multi_modal_autonomous_instruction_prompt import (
+    MULTI_MODAL_AUTO_AGENT_SYSTEM_PROMPT_1,
+)
 
-# Initialize the llm
 llm = GPT4VisionAPI()
 
-task = "Analyze this image of an assembly line and identify any issues such as misaligned parts, defects, or deviations from the standard assembly process. IF there is anything unsafe in the image, explain why it is unsafe and how it could be improved."
+task = (
+    "Analyze this image of an assembly line and identify any issues such as"
+    " misaligned parts, defects, or deviations from the standard assembly"
+    " process. IF there is anything unsafe in the image, explain why it is"
+    " unsafe and how it could be improved."
+)
 img = "assembly_line.jpg"
 
 ## Initialize the workflow
 flow = Flow(
     llm=llm,
-    max_loops=1,
+    max_loops='auto'
+    sop=MULTI_MODAL_AUTO_AGENT_SYSTEM_PROMPT_1,
     dashboard=True,
 )
 
-# Run the flow
 flow.run(task=task, img=img)
 
 
+
 ```
 
 ---
diff --git a/playground/demos/assembly/assembly.py b/playground/demos/assembly/assembly.py
index d3ebf39a..6722489a 100644
--- a/playground/demos/assembly/assembly.py
+++ b/playground/demos/assembly/assembly.py
@@ -1,6 +1,8 @@
 from swarms.structs import Flow
 from swarms.models.gpt4_vision_api import GPT4VisionAPI
-
+from swarms.prompts.multi_modal_autonomous_instruction_prompt import (
+    MULTI_MODAL_AUTO_AGENT_SYSTEM_PROMPT_1,
+)
 
 llm = GPT4VisionAPI()