Added battle.py