Improved comment UX