Is this not just giving `U[:, :n_eig]`?