Added a new function for custom color maps