Building training/test dataset

Building training/test dataset and visualization rutines