Add missing CSS classes