keeping now idxs for neighbourhood for writing colors back

by the way, maybe I can discard the neighbourhood earlier, I don't need it there anymore (probably)