added unit tests
added unit test