small Ui fixes