added base model support