Poprawki algorytmu i dodatkowy test