Added support for multi colored graphs