Minor edits in BattleScreen