Cleaning up the UI.