refactor sum.py