ui fixes
ui fixes