Fix/ui fixes