Update logistic_sgd.py