Added tree controller setup method to JSDocument