Update battle.py

No