added function to change color