Add task tests