Modified ui for demo