unit tests
Merge pull request #3 from meefik/unit-tests

Unit tests