Add max_dr, max_dist, which_mom.