diff --git a/uml/main.wsd b/uml/main.wsd
index 359690b..4ed71f7 100644
--- a/uml/main.wsd
+++ b/uml/main.wsd
@@ -1,7 +1,19 @@
 @startuml
 package Game {
     class Game {
-        + levels[]
+        + name: String
+        + levels[]: Level
+        + currentLevel: Level
+        
+        + changeLevel(Level lvl)
+    }
+    class Level {
+        + name: String
+        + gameObjects[]: gameObject
+        + load()
+        + start()
+        + draw()
+        + update()
     }
 }
 package Frontend {