add tests
Merge pull request #1 from jb55/tests

Add tests