Moved Probability annotation to probabilistic package