Add more tests

Ah.