Added class diagram for AIExtensions