z = np.real(np.dot(psi.conj(),np.dot(np.array([[1,0],[0,-1]]),psi)))