Add helper gradient function