small change to contour plot figure