Add unit tests
Merge pull request #3 from clarle/unit-tests

Add unit tests