Update diffie.py