In [5]:
def f(w):
    return (w-3)**2