1 files changed, 3 insertions, 0 deletions
diff --git a/Generator.h b/Generator.h
index 1679055..de3812a 100644
--- a/Generator.h
+++ b/Generator.h
@@ -8,11 +8,14 @@
 class Generator
 {
 public:
+    Generator();
     virtual ~Generator() = default;
     virtual void update(float dt) = 0;
+    void toggle(bool enabled);
     void attach(std::shared_ptr<Vector2> origin);
     void detach();
     Vector2 position() const;
 protected:
+    bool m_enabled;
     std::shared_ptr<Vector2> m_origin;
 };