Add the test framework