diff --git a/docs/assets/img/agent_def.png b/docs/assets/img/agent_def.png
new file mode 100644
index 00000000..c5ae1826
Binary files /dev/null and b/docs/assets/img/agent_def.png differ
diff --git a/docs/swarms/agents/toolagent.md b/docs/swarms/agents/toolagent.md
index c919866f..5a731e17 100644
--- a/docs/swarms/agents/toolagent.md
+++ b/docs/swarms/agents/toolagent.md
@@ -4,26 +4,6 @@ The `ToolAgent` class represents an intelligent agent capable of performing a sp
 
 The primary objective of the `ToolAgent` class is to amplify the efficiency of developers and AI practitioners by simplifying the process of generating meaningful outputs that navigate the complexities of the model and tokenizer.
 
-### Class Definition
-
-The `ToolAgent` class has the following definition:
-
-```python
-class ToolAgent(BaseLLM):
-    def __init__(
-        self,
-        name: str,
-        description: str,
-        model: Any,
-        tokenizer: Any,
-        json_schema: Any,
-        *args,
-        **kwargs,
-    )
-    def run(self, task: str, *args, **kwargs)
-    def __call__(self, task: str, *args, **kwargs)
-```
-
 ### Arguments
 
 The `ToolAgent` class takes the following arguments:
diff --git a/docs/swarms/framework/index.md b/docs/swarms/framework/index.md
index 0b298bf3..1331d935 100644
--- a/docs/swarms/framework/index.md
+++ b/docs/swarms/framework/index.md
@@ -17,17 +17,7 @@ Below are visual graphs illustrating the hierarchical and tree structure of the
 
 #### 1. Foundational Components: Models, Tools, Memory
 
-```mermaid
-graph TD;
-    Models --> Agents
-    Tools --> Agents
-    Memory --> Agents
-    subgraph Foundational_Components
-        Models
-        Tools
-        Memory
-    end
-```
+![Diagram](assets/img/agent_def.png)
 
 #### 2. Agents and Their Interactions