added probabilistic labeling