Working on the stage selector