Adjusted the rainbow function in the color shader