1 files changed, 14 insertions, 7 deletions
diff --git a/include/main.h b/include/main.h
index 6cc29c5..0d4de6f 100755
--- a/include/main.h
+++ b/include/main.h
@@ -7,16 +7,23 @@ typedef void (*StateFunc)(void);
 
 struct Main
 {
-    /*0x00*/u8 filler0[0x2];
-    /*0x02*/u8 mainState;
-    /*0x03*/u8 subState;
-    /*0x04*/u8 filler4[0x12];
-    /*0x16*/u16 unk16;
-    /*0x18*/u8 filler18[0x28];
-    /*0x40*/int unk40;
+    /*0x00*/ u8 filler0[0x2];
+    /*0x02*/ u8 mainState;
+    /*0x03*/ u8 subState;
+    /*0x04*/ u8 filler4[0x9];
+    /*0x0D*/ u8 unkD;
+    /*0x0E*/ u8 fillerE[0x8];
+    /*0x16*/ u16 unk16;
+    /*0x18*/ u16 newKeys;
+    /*0x1A*/ u16 releasedKeys;
+    /*0x1C*/ u16 heldKeys;
+    /*0x1E*/ u8 filler1E[0x22];
+    /*0x40*/ int unk40;
 };
 
 extern struct Main gMain;
 extern StateFunc gMainFuncs[];
 
+void SetMainGameState(u16);
+
 #endif // GUARD_MAIN_H