Diffie-Hellman Key Exchange Protocol in Python