def basic_calculator(a, b, op): """Perform a basic math operation.""" if op == "+": return a + b elif op == "-": return a - b elif op == "*": return a * b elif op == "/": if b == 0: raise ValueError("Division by zero") return a / b return None