Add some test
Merge pull request #1 from colinfrei/solution

Add some tests