Refactoring: use sum() instead of for loop

See for reference: https://docs.sourcery.ai/Reference/Python/Default-Rules/sum-comprehension/