Added AI.py, main is working