Update kl.py

prune unused code and comments