fix exercise 3