Added an apply_source and eliminated duplicated code for reduce_balanced