Create weighted_distribution.py