Bug fixes for cal_kl.py