added unit tests
added unit tests