Adding UI for demo