Calculator

def calculator(a, b, op='add'):
    if op == 'add':
        return a + b
    elif op == 'sub':
        return a - b
    else:
        print('Invalid operator')
        return None

calculator(1, 2, 'add')
calculator(1, 2, 'sub')
calculator(1, 2, '*')