Update rait.py