Communication in a distributed matrix + small changes