Style and UX updates