adding the colorful number code in python