Added support for lambda syntax for UnitOfWork