Update joint_prior.py