Moved batch unit storage to the mesh, improved locality