draw rainbow using python