Create probability.py