adding numpy array masking