Add code for generating validation dataset