Clean up Exercise 4