Переделал в генератор