* Add the option to handle depth for the tree flag * Update README to describe the new option for the tree flag * Add tests