Saving the mask image, still not correct