working on model1 and different datasets