Add minor changes/comments to UI.hs