Clean up node colouring