Add setColorMode function