Added the task model
seems to be working well :)