moved computation of KL divergence to support
increased speed of visualizer reusing already computed values
added exception handling for bfgs