add set_mag method