Remove computation of the 2-norm of the gradient