Correct solution for problem 1.