Update battle.cpp