problems with training and testset