Add tests

Closes #3.