Python Function With Multiple Arguments Using If Statements