Add digit sum function