Finish problem 2 except answer