add pick_sample too