Add LR sgd code