Fixed the documentation for the example