public virtual T Add<T>(T entity) where T : class
        {
            entity = Set<T>().Add(entity);
            int res = SaveChanges();

            return entity;
        }