Updated stage demo with stage changes