Update log_sum.py