Adjust iterator for neighbour loops and fix sum_k_grad_Ci calculation.