Added setColor function