Add changeColor function