Function With Argument In Python