Modified the training/validation set
Removed some outputs