add a visualization for the mask