Add JS to change color