small changes
Added some tests!
Merge pull request #7 from ianbishop/example-tests

Added some tests!