diff --git a/uml/main2.wsd b/uml/main2.wsd
index 3ff1a33..e7ae8a1 100644
--- a/uml/main2.wsd
+++ b/uml/main2.wsd
@@ -9,7 +9,8 @@ package Game {
     }
     class Level {
         + name: String
-        + gameObjects[]: gameObject
+        + gameObjects[]: GameObject
+        + inputs[]: Input
         + load()
         + start()
         + draw()
@@ -23,6 +24,7 @@ package Game {
         + y: double
         + width: double
         + height: double
+        + rotation: double
         + img: Image
         + layer: int
         + load()
@@ -31,6 +33,12 @@ package Game {
         + draw()
         + update()
     }
+    class Input {
+      + key: String
+      + type: String
+      + listen()
+      + action()
+    }
 }
 package Frontend {
 
@@ -53,6 +61,7 @@ json MainGameJSON{
     "Y":0,
     "Width":100,
     "Height":100,
+    "Rotation":90,
     "imgSRC":"Image",
     "layer":1