Minor fix to SGD update formula.