Add a simple test suite