1 files changed, 19 insertions, 2 deletions

diff --git a/src/game.c b/src/game.c
index a093269..899b3f2 100644
--- a/src/game.c
+++ b/src/game.c
@@ -3,19 +3,36 @@
 void initGame(Game* game)
 {
     InitWindow(WINDOW_WIDTH, WINDOW_HEIGHT, "Penguin Yippies!");
+
+    // Screens.
+    game->currentScreen = MAIN_MENU_SCREEN;
+    initMainMenu(&game->mainMenu, game);
+    initGameScreen(&game->gameScreen, game);
 }
 
 void updateGame(Game* game)
 {
     BeginDrawing();
-    ClearBackground(RAYWHITE);
 
-    DrawFPS(0, 0);
+    switch (game->currentScreen)
+    {
+        case MAIN_MENU_SCREEN:
+            updateMainMenu(&game->mainMenu, game);
+            break;
+        case GAME_SCREEN:
+            updateGameScreen(&game->gameScreen, game);
+            break;
+        default:
+            break;
+    }
 
     EndDrawing();
 }
 
 void closeGame(Game* game)
 {
+    closeMainMenu(&game->mainMenu);
+    closeGameScreen(&game->gameScreen);
+
     CloseWindow();
 }