Оптимизация функции подсчёта суммы

Улучшить читабельность кода. Убрать неэффективные конструкции.
Проверить, нет ли ошибок или потенциальных проблем.

def```