Adding a colormap legend to a figure