add test that kl is correct on simple example