Working on problem 11