Working on problem 7