finish problem 2
finish problem 2