Add some tests
Merge pull request #10 from summeragain/master

Add some tests