Adding color mapping