Add a little test suite