Added more tests for task model