Modified addTrain Method