Кто может помочь добавить изменение в код python?

LVitA, а он и написал, как надо. А надо взять и разобраться самостоятельно. Вот такой ответ и был.