Update exercise 2