Fix exercise 001