updates for exercise 3