Improved AI performance