1 files changed, 7 insertions, 2 deletions
diff --git a/src/screens.h b/src/screens.h
index 75e9616..f73df4e 100644
--- a/src/screens.h
+++ b/src/screens.h
@@ -1,6 +1,11 @@
 #include <raylib.h>
 
 #include "editor.h"
+#include "xd.h"
 
-void SE_DrawMenuScreen();
-void SE_DrawEditorScreen(Editor *editor, Camera2D *camera);
+typedef enum GameScreen { SCREEN_MENU = 0, SCREEN_EDITOR } GameScreen;
+
+void SE_DrawMenuScreen(GameScreen *currentScreen, Editor *editor, XdData *data,
+                       XdData *datas[]);
+void SE_DrawEditorScreen(GameScreen *currentScreen, Editor *editor,
+                         Camera2D *camera);