Added a change color function