Added the model to the task

Added the model to the task