problem 3 works.