Dodałam zmiany w funkcji liczącej sąsiadów