1 files changed, 3 insertions, 2 deletions

diff --git a/src/docs/class.puml b/src/docs/class.puml
index f4766db..30d50f2 100644
--- a/src/docs/class.puml
+++ b/src/docs/class.puml
@@ -20,6 +20,7 @@ package utils {
 package viewer {
   class Viewer {
     Game
+    Stage
     void main(String[])
     void eventLoop(ApplicationContext)
     void renderFrame(Graphics2D)
@@ -55,8 +56,8 @@ package context {
 
   class Game {
     final GameController
-    Stage
-    Game(GameController,Stage,List<Board>)
+    final List<Board>
+    Game(GameController,List<Board>,List<Board>)
   }
 }