Add code to generate plots
Add output plots