Added support for SGD

Lets work together :)
as above