def foo(x):
    return x**2 + 2*x + 1