Updating docs to include context manager example.