Create probability_1.py