It would be visually helpful if when you select a stage, all the child components of that stage would also become selected in the rocket design view. Same could be done for selecting the rocket: select all components in the design view.

Demo: