Adding new feature: color mapping