Python Write A Function