Get battle mode working