add highlight-stages

please add highlight-stages