add tests
Merge pull request #2 from bcherny/add-tests

add tests