1 files changed, 10 insertions, 6 deletions
diff --git a/Falling.cpp b/Falling.cpp
index deb81a4..38c05d4 100644
--- a/Falling.cpp
+++ b/Falling.cpp
@@ -1,20 +1,24 @@
 #include "Falling.h"
 
+#include <memory>
+
 #include <raylib.h>
 
 #include "Generator.h"
 
 
-Falling::Falling() :
-    m_speed {40}
+Falling::Falling(std::shared_ptr<Vector2> position, std::shared_ptr<Generator> generator) :
+    m_speed {40},
+    m_position {position},
+    m_generator {generator}
 {
 }
 
 
 void
-Falling::update(const float dt, Vector2& position, Generator& generator)
+Falling::update(const float dt)
 {
-    position.y += dt * m_speed;
-    if (position.y > 600)
-        generator.toggle(false);
+    m_position->y += dt * m_speed;
+    if (m_generator->m_enabled && m_position->y > 620)
+        m_generator->m_enabled = false;
 }