Different aproach to BattleUnit class