Improve some UI stuff.