changed to 2 dimension grid to predict labels. start adding comments to kernels