Add some boilerplate for test runner