Modeling and control of a water treatment plant