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