Added battle button and logic