Add some simple tests

There's more to be done, but it's a start