add set_edge_color function