Updated the base model