Added battle.py

Added battle.py