diff --git a/README.md b/README.md
index 36275b5d516c7a98f56fa2551d66c2b1c3aecf17..027f912973af6c884a6927e104cfb40421aa0a98 100644
--- a/README.md
+++ b/README.md
@@ -15,6 +15,10 @@
 * It lets you build LLM-based agents from graphs.
 * It enables the customized and automatic self-organization of agent swarms with self-improvement capabilities.
 
+## Visualizations of the graphs
+
+<img src="assets/swarm_vis.png" alt="Edge optimization" width="800">
+
 ## News
 
 * 🔥 **[07/05]** Mingchen gave an invited talk to WAIC.
@@ -126,9 +130,6 @@ Here is a Youtube video on how to run the demo notebooks:
 
 <img src="assets/swarm_v3.png" alt="Edge optimization" width="500">
 
-## More Visualizations
-
-<img src="assets/swarm_vis.png" alt="Edge optimization" width="800">
 
 ## Running with a local LLM