Final changes made to AI component