Water treatment for chiller systems