1 files changed, 3 insertions, 1 deletions
diff --git a/src/entity.hpp b/src/entity.hpp
index 6c803df..40818ee 100644
--- a/src/entity.hpp
+++ b/src/entity.hpp
@@ -12,6 +12,7 @@
 #include "window.hpp"
 
 #include "weapon.hpp"
+#include <map>
 
 class Entity {
     public:
@@ -25,7 +26,8 @@ class Entity {
         double furyTime;
         double reloadTime;
         Vector2 direction;
-        AWeapon * wp;
+        std::map<int, AWeapon*> wp;
+        AWeapon * currentWeapon;
         Image img;
         Texture2D idleTex;
         Texture2D hurtTex;