add update function in rokko/distributed_matrix.hpp