Update units.py to include 'mag' units