adding support to tap into stage choices and challenge stages