Improved example

I updated the images in the doc. I also improved the example by adding some functionality and a few comments after our mail conversation.