Add --tree-depth CLI argument