Adds some tests to prove it works