added simple unit testing