now backprop for grad validates, need to optimize it