Spruced up some of the UI