Add tests

Now, with tests!