Started working on battle mode