Added support for SGD

Lets work together :)