diff --git a/README.md b/README.md
index a2c89e4..008d2f5 100644
--- a/README.md
+++ b/README.md
@@ -75,16 +75,19 @@ actor "Player 1" as P1
 actor "Player 2" as P2
 actor "Player 3" as P3
 
+
 package Masterpiece {
-    usecase Game as "FairHopper\nGame Server"
-    usecase WS as "WS Server"
+    rectangle {
+        usecase Game as "FairHopper\nGame Server"
+        usecase WS as "WS Server"
+    }
     usecase Vis as "Visualisation\nService"
 }
 
 P1 -left-> Game: REST API
 P2 -left-> Game: REST API
 P3 -left-> Game: REST API
-Game --> WS: WebSockets
+Game --> WS: Game State
 WS --> Vis: WebSockets
 ```