add max depth for a decision tree