arg_z = (x**2 + y**2)