Python Create Function With Arguments And Return