add arxiv kl stuff