1 files changed, 3 insertions, 5 deletions
diff --git a/Unit.h b/Unit.h
index 55439c4..eecdbcd 100644
--- a/Unit.h
+++ b/Unit.h
@@ -77,7 +77,8 @@ typedef enum t_availableactions
 {
     ACTION_BUILD,
     ACTION_ATTACK,
-    ACTION_CREATE_UNIT
+    ACTION_CREATE_UNIT,
+    MAX_ACTIONS
 }UNIT_ACTION;
 
 typedef struct t_Camera TYPE_CAMERA;
@@ -88,7 +89,7 @@ typedef struct t_Camera TYPE_CAMERA;
 
 // Initialization and handling
 void        UnitInit(void);
-void        UnitHandler(TYPE_UNIT* ptrUnit);
+void        UnitHandler(TYPE_UNIT* unitArray, size_t sz);
 
 // Unit information
 uint8_t     UnitGetHpFromID(uint8_t id);
@@ -101,9 +102,6 @@ void        UnitDraw(TYPE_UNIT* ptrUnit, TYPE_CAMERA* ptrCamera, bool bHighlight
 
 void        UnitMoveTo(TYPE_UNIT* ptrUnit, uint16_t x, uint16_t y);
 
-void        UnitAcceptAction(TYPE_UNIT* ptrUnit);
-void        UnitResetMenuLevel(void);
-
 // Selection index
 const char* UnitGetActionString(UNIT_ACTION action);