Optimize the following piece of code written in Python
def add(a, b):
    c = a + b
    return c
def add(a, b):
    return a + b  # remove unnecessary intermediate variable 'c' and return the result directly.