Added unit tests
Merge pull request #2 from EduardTrayan/add-unitests

Added unit tests