updates to the UI

ui fixes