addition to the "similarity" part of the masking section and first strides in the "uncertainty" section.