working on gradient updates