Wastewater plant upgrades