using get_models to get the base models