Add tests

I just don't understand what you're trying to achieve.