refactor code to random forest and add decision tree