Update ChooseStage.java