Added tests for sum_of_digits