Improve the example plot and add it to the readme