Add option to return tree starting a specified node