Decision tree and random forest project