Poprawa algorytmu