A Function With An Input Argument Python