Update test framework for testing