Added plotting of results.