Update example in README and docs to match context manager changes