update ai.py