Fixed some problems in UI.