minor fixes on the UI