Update docs in remove_masked_segments