Fixed exercise 4