Add tests

Fixes #26