Add tests

Obviously ..