adding some changes to test runner