Ui fixes/adjusments