z2 = np.exp(-(xx - 1)**2 - (yy - 1)**2)