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

Add tests