Python Function With Optional Input