Remove old procedure on probabilities and add folds