Add tests

This is important.