Updated TODO in weighted_sum.v