finish problem 2