log_sum_exp added