add stage switch to challenge