Got AI transition working