Add Exercise 4 Code