How To Put Input In A Function Python