Add task 1 from the first problem set