Add comment update feature