new unit tests for the new classes