Final changes to AI