Added battle functionality