Add tests

Because every gem needs tests!