Update mag_diff.py