Colour Game In Python Code