Update ks.py