Update rafiki.py