Added code to plot the data