Update polygon_plot.py