Update ch01_fig1-3_contour.py