Working on problem 2