added unit test class