Being used as a processor makes so much more sense than a battery. I suppose they changed it to make it seem more degrading?
The battery thing seemed so stupid to me.
Also, there's the line from Smith that they tried making it a utopia but people worked better when they were miserable, or whatever he said.