Update logistic_model.py