revert to label sampling  and add pickled sample