Update digit_sum.py