little ui fixes