Change TestSetBuilder.create to accept function that will call test function