Toggle color change is almost working