Adding checks/fleshing out distributed_flow