add tests
Merge pull request #6 from JayDevlin/add-tests

add tests