adding support for running different datasets