Function With Input In Python