Change soft_mask example to be representative