Updated Exercise 4