Move color logic into Node class