problem 11 finally correct