Added a gradient to the mass