Added plotting for entropy and divergence