Committing correction to the 4th code and plot