Added a base model and use it on task model