cleaning up the complexities from Class 4 Exercise 5