added stage selection