Fixed updateColors function