Slight modification to Problem 9