Add code to plot the data

The code was taken from the PDF.

Such code
                wow!
                                so plotting
     what plotting

                very machine learning