Data augmentation

... still writing on it