Updating exercise 3