Put in ability to remove a test from a class