Update diagram with better high-lights.