BattleScreen class added