Water treatment process optimization