SGD is not really working.. so modified the write up to say SGD or mini batch