fixed bug with kl_div
removed debug output