some small ui fixes