Доработка генератора