Cleanups to the UI.