Refactored enrichment CSS class setting.