Updating exercise 4