Added contour plot function