Function for scaling test set