updating sgd.py