Added code to create plots.