UI Improvements to the Demo.