battle.py updated