Problem 9 - correct a type