Improve documentation in example