Improve node coloring (remove the slight delay)