Adding code for random forest