extended the batch layer