Updated the example and documentation