Add some tests

Fixes #1