added held-out state marginals