1 files changed, 5 insertions, 2 deletions
diff --git a/game/entity.h b/game/entity.h
index eeab110..50c7acc 100644
--- a/game/entity.h
+++ b/game/entity.h
@@ -16,6 +16,7 @@ class EntityInfo {
     std::string name;
     
     int id;
+    int level;
     
     float hp;
     float attack;
@@ -28,11 +29,13 @@ class EntityInfo {
     std::vector<int> build;
     std::vector<int> train;
     
-    std::vector<int> prod { 0, 0 };
-    std::vector<int> cost { 0, 0 };
+    std::vector<float> prod { 0, 0 };
+    std::vector<float> cost { 0, 0 };
     
     std::vector<int> adjacent;
     
+    int upgrade { -1 };
+    
     enum Class { inf, cav, ran, sie, bld };
     Class ent_class;