Update diffie-hellman.py