def operation(number1, number2, operator): if operator == '+': return number1 + number2 elif operator == '-': return number1 - number2