Adding a few more tests. Fixes #7.