Update logit.py