Add tests
Merge pull request #4 from jeffjo/jeffjo/add-tests

Add tests