Added unit test
added unit test