diff --git a/README.md b/README.md
index 008d2f5..0a5cf7f 100644
--- a/README.md
+++ b/README.md
@@ -94,30 +94,26 @@ WS --> Vis: WebSockets
 ### WebSockets
 
 ```plantuml
-    participant Game as "FairHopper\nGame Server"
+box "FairHopper Game Server" #lightcyan
+    participant Game as "Game Engine"
     participant WS as "WS Server"
-    participant Client1 as "Visualisation\nClient 1"
-    participant Client2 as "Visualisation\nClient 2"
+endbox
+participant Client1 as "Visualisation\nClient 1"
+participant Client2 as "Visualisation\nClient 2"
 
-    Game ->o WS: Server Connect
-    activate WS #coral
-        WS -> Game: Get game state
-        activate Game #yellow
-            Game -> WS: Game state
-        deactivate
-    deactivate
+Game ->o WS: Send initial state
 
-    Client1 ->o WS: Client Connect
-    activate WS #coral
-        WS -> Client1: Game state
-    deactivate
+Client1 ->o WS: Client connect
+activate WS #coral
+	WS -> Client1: Game state
+deactivate
 
-    Client2 ->o WS: Client Connect
-    activate WS #coral
-        WS -> Client2: Game state
-    deactivate
+Client2 ->o WS: Client connect
+activate WS #coral
+	WS -> Client2: Game state
+deactivate
 
-    loop #lightyellow On game state change
+loop #lightyellow On game state change
     Game ->o WS: Game state
     activate WS #coral
         WS o-> Client1: Game state