replaced test dataset with all data