Modify auth.py