Added function to create contour plots