Removed comments in ai.py