Add more tests

Just define the implementation a little.