Update UI to show hide correct number of stages