Add loop to catch color changes