Create diffie.py