- added unit tests