Update battle.py

Finish update.