Create battle.py