Dostosowanie klasy Generator do zadania 2.