Add more docs for UnitOfWork internals