Add missing CSS class