Ui

Minor UI cleanup.