added tests to the unit tests