update question 2 in exercise 4