Improve comment UI style