Parallelized shallow_water_extension