Add option to toggle tree rendering