Added Battle class