Made some small improvements to the UI.