max_depth = 2