Add Battle class for the actual battle