added simple unit tests