Added remove comment functionality

Minor fix to message hiding