add Battle class