Update BattleDriver.java