Model of wastewater treatment