renamed 'featx' to 'ai_module'