ui fixes and updates