Improved the look of the ui.