add tests
Merge pull request #4 from JasonBoyles/add-tests

add tests