poprawki algorytmu + logi