added python function for vertex coloring