adding color map functionality