Update to add color to nodes