Misc UX tweaks