Add battle logic