Change color to index-based