Update digit_sum.py

added repeat option along with a small fix