add wrapper to build training, test and val sets