added unit tests
added comment