Update gradient.cpp