Update longitudinal_distributions.py