modify exercise 3