Add the ability to define cleanup for a test