Add tests

Because every :gem: needs tests!