Некоторые правки по генерации. Добавил новые тесты