Modify init, setStage method. Refined demo page.