1 files changed, 3 insertions, 2 deletions

diff --git a/src/docs/class.puml b/src/docs/class.puml
index 7a00135..754ddbe 100644
--- a/src/docs/class.puml
+++ b/src/docs/class.puml
@@ -11,6 +11,7 @@ package utils {
 package viewer {
   class Viewer {
     Game
+    Stage
     void main(String[])
     void eventLoop(ApplicationContext)
     void renderFrame(Graphics2D)
@@ -46,8 +47,8 @@ package context {
 
   class Game {
     final GameController
-    Stage
-    Game(GameController,Stage,List<Board>)
+    final List<Board>
+    Game(GameController,List<Board>)
   }
 }