Parameterize the max depth.