Update kai.py