Running with different Datasets - Need to finetune performance