Update Battle.py