Add tests
Merge pull request #3 from mirego/feature/tests

Add tests