1 files changed, 5 insertions, 5 deletions
diff --git a/ExampleGenerator.h b/ExampleGenerator.h
index 0e3372b..25c4ce3 100644
--- a/ExampleGenerator.h
+++ b/ExampleGenerator.h
@@ -1,16 +1,16 @@
 #pragma once
 
-#include <vector>
-
 #include <raylib.h>
 
 #include "Bullets.h"
+#include "Generator.h"
 
 
-struct ExampleGenerator
+struct ExampleGenerator : public Generator
 {
-    ExampleGenerator();
-    void update(float dt, std::vector<ConstantVelocityBullet>& bullets);
+    explicit ExampleGenerator(ConstantVelocityBullet::Vector& bullets);
+    void update(float dt) override;
+    ConstantVelocityBullet::Vector& m_bullets;
     float m_delay;
     float m_interval;
     float m_cone;