Update distributions.py