Adding feature for edditing comments