Add CSS classes to handle node colors