Fixed a few issues in the UI.