Diffie-Hellman key exchange example in python