Added logic for color-changing background