Cleanup the UI a bit.