Initial support for AI working with UI