Чистый код и тестирование

Оформить и переработать следующий код, приведя его к требованиям чистого кода:

• Добавить аннотации типов, где их не хватает.

• Добавить docstring