improve comment UI