diff --git a/docs/visualizations.md b/docs/visualization.md
similarity index 97%
rename from docs/visualizations.md
rename to docs/visualization.md
index 2a8bb28..6bb5ee5 100644
--- a/docs/visualizations.md
+++ b/docs/visualization.md
@@ -7,7 +7,7 @@ Agent visualization allows you to generate a structured graphical representation
 The visualization functionality relies on the **Graphviz** package. To use it, ensure you have Graphviz installed and add it as a dependency in `pyproject.toml`. Alternatively, install it directly via pip:
 
 ```bash
-pip install graphviz
+pip install openai-agents[visualization]
 ```
 
 ## Generating a Graph
@@ -22,7 +22,7 @@ You can generate an agent visualization using the `draw_graph` function. This fu
 
 ```python
 from agents import Agent, function_tool
-from agents.visualizations import draw_graph
+from agents.extensions.visualization import draw_graph
 
 @function_tool
 def get_weather(city: str) -> str: