Fix exercise 1