some ui fixes
UI changes