Update battleEngine.py