Add context manager for connection pool for use with the with statement