Implemented tests for the classes outlined in the exercises