Better way to grid search max_depth in example