add comments in plot_array.py