Add support for coloring vertices