Finish problem 2