z = np.mean(samples[:,2]) * np.array([[1,0],[0,1]])