1 files changed, 6 insertions, 2 deletions
diff --git a/Falling.h b/Falling.h
index b629d6d..b50228a 100644
--- a/Falling.h
+++ b/Falling.h
@@ -1,5 +1,7 @@
 #pragma once
 
+#include <memory>
+
 #include <raylib.h>
 
 #include "Behaviour.h"
@@ -9,8 +11,10 @@
 class Falling : virtual public Behaviour
 {
 public:
-    Falling();
-    void update(float dt, Vector2& position, Generator& generator) override;
+    Falling(std::shared_ptr<Vector2> position, std::shared_ptr<Generator> generator);
+    void update(float dt) override;
 private:
     float m_speed;
+    std::shared_ptr<Vector2> m_position;
+    std::shared_ptr<Generator> m_generator;
 };