minor change on exercise 9