Battle class added