Update jai.py