Added function to plot and compare.