homework 4: colormap function