Update Abby's code. Finish reduce communication