Adding two classes from the example and challenge from chapter 18