Add training code for rpart and knn; update readme