Add ability to add custom color map