Updated example and documentation