This is the edited UI and will be used in the demo