water treatment plant upgrades