Added rainbow mode toggling code