Add task_3 and tests