Update K2.py