- added unit test