clarification: change masks to class_membership_masks