Improved AI behavior