Improved the code in SceneStage.

[ENH] Improved the code in SceneStage.