Add plotting of results