Added battle class