Added the colormap and normalize as arguments