adding a max_depth to the tree printing