Add tests
Merge pull request #2 from mirego/feature/add-tests

Add tests