Water treatment in power plants