Added color mapping implementation