Minor updates to the documentation of the example