Update exercise 4