Update ai.py
Merge pull request #6 from SirRainy/patch-4

Update ai.py