Fix m2_exchange_flux function so that magnetic flux is updated on shells