Fix exercise 4