Refine the tag close button and styling