Adding unit tests for new classes.