small ui fixes