diff --git a/docs/building-with-codegen/codebase-visualization.mdx b/docs/building-with-codegen/codebase-visualization.mdx
index 64a2ea240..0a4c7c70a 100644
--- a/docs/building-with-codegen/codebase-visualization.mdx
+++ b/docs/building-with-codegen/codebase-visualization.mdx
@@ -14,11 +14,22 @@ These visualizations have a number of applications, including:
 - Analyzing dependencies
 - Understanding inheritance hierarchies
 
-This guide provides a basic overview of graph creation and customization.
+This guide provides a basic overview of graph creation and customization. Like the one below which displays the call_graph for the [modal/client.py](https://github.com/modal-labs/modal-client/blob/v0.72.49/modal/client.py) module.
+
+<iframe
+  width="100%"
+  height="600px"
+  scrolling="no"
+  src={`http://localhost:3000/embedded/graph?id=ef312926-286d-4c13-8e09-38e1a47882f1&zoom=0.5`}
+  className="rounded-xl "
+  style={{
+    backgroundColor: "#15141b",
+  }}
+></iframe>
 
 <Note>
   Codegen visualizations are powered by [NetworkX](https://networkx.org/) and
-  rendered using [ThreeJS](https://threejs.org/).
+  rendered using [d3](https://d3js.org/what-is-d3).
 </Note>
 
 ## Basic Usage
diff --git a/docs/introduction/overview.mdx b/docs/introduction/overview.mdx
index d386f84e0..da55ea2b4 100644
--- a/docs/introduction/overview.mdx
+++ b/docs/introduction/overview.mdx
@@ -36,9 +36,11 @@ function.set_docstring('new docstring') # set docstring
   src={`https://chadcode.sh/embedded/codemod/?code=${encodeURIComponent(
     intoSnippet
   )}`}
-  className="rounded-md "
-
-> </iframe>
+  style={{
+    backgroundColor: "#15141b",
+  }}
+  className="rounded-xl"
+></iframe>
 
 ## Installation