Add the code and plot for the second part or the project