Excercise 3: fix