Add tests.
Merge pull request #4 from macejmic/tests

Add tests.