Get battle termination logic working