Updated to problem 9