Modify label_binarize