Optimize kn.py