Update example_ai.py