enhance ai module #2