Fix rainbow demo script