Add classes UnitOfWOrk, QueryBuilder