add stage-display code