Minor revision to battle_engine.py