Create UnitOfWork class