modyfing UI for demo