Improved code for restoring light colors after canceling