UI

ui fixes