Added gradient regularization