Amendments made to battle.py