update course project: use rpart in train()