Added remove option to update op