Create calculator

def calculator(a, b, op='add'):
  if op == 'add':
    return a + b
  elif op == 'sub':
    return a - b
  else:
    print('Valid operations are add and sub')

calculator(1, 2, 'add')