Update collections_original_probs.py