Create digit_sum.py