Added support for SGD

Lets work together :)
again, as above