Extend the test framework