updated ai_scheme