It would be great if this can be refactored to `AWSBaseConnection` class and re-used with other drivers.