Adding random object query functionality to UnitOfWork