def color(word, letter, idx): cond: word[idx] == letter: return GREEN letter in word: return YELLOW True: return GREY