VERSION 5:
simplify the 'digit_sum' function