Brush up code
Add getNextBattleAction