Cleaning up and separating contour plot example