Update ai.py