Codecademy Python3 Code Challenge: Count First Letter

# Write your count_first_letter function here:
def count_first_letter(names):
  letters = {}

  for key, values in names.items():
    letter = key[0]
    if letter not in letters:
      letters[letter] = 0

    if key[0] == letter:
      letters[letter] += len(values)


  return letters

# Uncomment these function calls to test your  function:
print(count_first_letter({"Stark": ["Ned", "Robb", "Sansa"], "Snow" : ["Jon"], "Lannister": ["Jaime", "Cersei", "Tywin"]}))
# should print {"S": 4, "L": 3}
#print(count_first_letter({"Stark": ["Ned", "Robb", "Sansa"], "Snow" : ["Jon"], "Sannister": ["Jaime", "Cersei", "Tywin"]}))
# should print {"S": 7}



#############
OFFICIAL:
#############

def count_first_letter(names):
  letters = {}
  for key in names:
    first_letter = key[0]
    if first_letter not in letters:
      letters[first_letter] = 0
    letters[first_letter] += len(names[key])
  return letters