Add enough code to run tests