Update general_2D_plot.py to support scatter plots and reuse ax to plot multiple data
mentioned in commit 57f3bd43
merged