1 files changed, 8 insertions, 1 deletions
diff --git a/Enemy.h b/Enemy.h
index 0a8c429..30e4140 100644
--- a/Enemy.h
+++ b/Enemy.h
@@ -4,18 +4,25 @@
 
 #include <raylib.h>
 
+#include "Behaviour.h"
 #include "Generator.h"
 
 
 class Enemy
 {
 public:
-    explicit Enemy(std::unique_ptr<Generator> generator = {});
+    explicit Enemy(std::unique_ptr<Generator> generator);
+    Enemy(
+        std::shared_ptr<Vector2> position,
+        std::unique_ptr<Generator> generator,
+        std::unique_ptr<Behaviour> behaviour);
     void update(float dt);
     void draw();
     void set_position(float x, float y);
+    void set_hold(float seconds);
 private:
     float m_hold;
     std::shared_ptr<Vector2> m_position;
     std::unique_ptr<Generator> m_generator;
+    std::unique_ptr<Behaviour> m_behaviour;
 };