Added support for new AI