Added __enter__ to uows client