changes to compare_masses.py