Gradient calculation now working
Calculate components with grad_u[i]