update for exercise 4