added unit test
Merge pull request #10 from cgoldberg/unit-tests

added unit test