create a rainbow in python