Automation of a water treatment system