Optimizing water infrastructure