z1 = np.exp(-xx**2 - yy**2)