def METHOD_NAME():
    plot = openmc.Plot()
    plot.color_by = 'material'
    plots = openmc.Plots([plot])
    model = openmc.examples.pwr_pin_cell()
    mats = {m for m in model.materials if 'UO2' in m.name}
    plots.highlight_domains(model.geometry, mats)