Add a function that allows us to query a colored set of cells.

This function is called on a separate task upon calling distribute_dofs() to run
in the background for when we need the result at a later time.
Update documentation in a couple of places.