Added code and images of the plot