Plots and code

Added required plots and code