ui fixes