add function to change color