1 files changed, 11 insertions, 1 deletions
diff --git a/Generator.h b/Generator.h
index 5beef2c..1679055 100644
--- a/Generator.h
+++ b/Generator.h
@@ -1,8 +1,18 @@
 #pragma once
 
+#include <memory>
 
-struct Generator
+#include <raylib.h>
+
+
+class Generator
 {
+public:
     virtual ~Generator() = default;
     virtual void update(float dt) = 0;
+    void attach(std::shared_ptr<Vector2> origin);
+    void detach();
+    Vector2 position() const;
+protected:
+    std::shared_ptr<Vector2> m_origin;
 };