1 files changed, 8 insertions, 2 deletions
diff --git a/ExampleGenerator.h b/ExampleGenerator.h
index 2710d03..6db4151 100644
--- a/ExampleGenerator.h
+++ b/ExampleGenerator.h
@@ -1,15 +1,21 @@
 #pragma once
 
+#include <memory>
+
 #include <raylib.h>
 
 #include "ConstantVelocity.h"
 #include "Generator.h"
 
 
-struct ExampleGenerator : public Generator
+class ExampleGenerator : public Generator
 {
-    explicit ExampleGenerator(ConstantVelocityBullet::Vector& bullets);
+public:
+    friend class EnemyFactory;
+    ExampleGenerator(std::shared_ptr<Vector2> position, ConstantVelocityBullet::Vector& bullets);
     void update(float dt) override;
+private:
+    std::shared_ptr<Vector2> m_position;
     ConstantVelocityBullet::Vector& m_bullets;
     float m_delay;
     float m_interval;