AI Program in Python