Update plot_ellipse.py