Use contextmanager for the transaction code samples and documentation

[mdumandag]

We have support for the contextmanager for the transactions but we are not using it for our code samples or documentation.

We should be using it to make it more readable and familiar with Python developers.