Improving example in `while` docs.