Add test. fixes #2
Merge pull request #5 from one-love/feature/add-test

Add test. fixes #2