Added the base class model