Add task tests!