Adjusted UI for demo