working on problem 2