data augmentation: new layer to load and augment and fixed some bugs