Add tests.
Merge pull request #3 from sorribas/tests

Add tests.