Updating exercise 2