Polish for stages in quest-UI