test set, better visualisation