edit_mask function

Edit mask