Update logistic_regression.py