1 files changed, 5 insertions, 7 deletions
diff --git a/GameScreen.h b/GameScreen.h
index 11d5b73..5b43512 100644
--- a/GameScreen.h
+++ b/GameScreen.h
@@ -1,21 +1,19 @@
 #pragma once
 
+#include <memory>
+
 #include <raylib.h>
 
-#include "ConstantVelocity.h"
-#include "ExampleGenerator.h"
-#include "Player.h"
 #include "Screen.h"
+#include "Stage.h"
 
 
 class GameScreen : public Screen
 {
 public:
-    GameScreen();
+    explicit GameScreen(std::unique_ptr<Stage> stage);
     void update(float dt) override;
     void draw() override;
 private:
-    Player m_player;
-    ConstantVelocitySystem m_const;
-    ExampleGenerator m_generator;
+    std::unique_ptr<Stage> m_stage;
 };