update to exercise 3