Ui improvements

Several visual improvements to the UI