How To Input A Function In Python