Adding the Batch class to GPU