Update array_plot_data.py