Update pot_dist.py