removed probabilities from main line code