Implemented simple decision tree model