Small additions to UI.