Update battle.py