Feature/task tests