Python colour recognition game