KeyError that I can't figure out. Moving on to problem3 for now.