Update example code in documentation.