Add set_color to mask broken pixels