changes for test dataset