plot_data code