Added in new base model